package org.example.test;

import java.lang.management.MemoryUsage;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Formatter;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

import javax.management.MBeanAttributeInfo;
import javax.management.MBeanInfo;
import javax.management.MBeanServerConnection;
import javax.management.ObjectInstance;
import javax.management.ObjectName;
import javax.management.openmbean.CompositeDataSupport;
import javax.management.remote.JMXConnector;
import javax.management.remote.JMXConnectorFactory;
import javax.management.remote.JMXServiceURL;
//
///**
// * JMX 获取 tomcat 监控信息
// * 首先可以用JDK提供的工具jconsole来供我们查看，这个在服务端配置下catalina.bat文件，加入以下内容：
// *
// * <p>
// * Linux下要监控的java进程需要配置参数： -Djava.rmi.server.hostname=192.168.1.126 #
// * Linux主机的ip，在windows输入的ip地址 -Dcom.sun.management.jmxremote
// * -Dcom.sun.management.jmxremote.port=7999 # 通信的端口，windows下输入
// * -Dcom.sun.management.jmxremote.authenticate=false # 不需要用户名，密码登录
// * -Dcom.sun.management.jmxremote.ssl=false
// * 在windows下，打开jconsole或jvisualvm，输入192.168.1.126:7999，不需要用户名密码直接链接即可。
// * </p>
// *
// * 端口7999可以自由更改，前提是没被系统占用。
// *
// *
// * @author Frank Cheung<sp42@qq.com>
// *
// */
public class JMX_Tomcat {
	static String jmxURL = "service:jmx:rmi:///jndi/rmi://192.168.1.126:7999/jmxrmi";// tomcat jmx url

	public static void start() {
		try {
			JMXServiceURL serviceURL = new JMXServiceURL(jmxURL);
			Map<String, String[]> map = new HashMap<>();
			map.put("jmx.remote.credentials", new String[] { "monitorRole", "QED" });

			try (JMXConnector connector = JMXConnectorFactory.connect(serviceURL, map);) {
				MBeanServerConnection mbsc = connector.getMBeanServerConnection();

				// 端口最好是动态取得
				ObjectName threadObjName = new ObjectName("Catalina:type=ThreadPool,name=http-8089");
				MBeanInfo mbInfo = mbsc.getMBeanInfo(threadObjName);

				String attrName = "currentThreadCount";// tomcat的线程数对应的属性值
				MBeanAttributeInfo[] mbAttributes = mbInfo.getAttributes();
				System.out.println("currentThreadCount:" + mbsc.getAttribute(threadObjName, attrName));

				// heap
				for (int j = 0; j < mbsc.getDomains().length; j++)
					System.out.println("###########" + mbsc.getDomains()[j]);

				Set<ObjectInstance> MBeanset = mbsc.queryMBeans(null, null);
				System.out.println("MBeanset.size() : " + MBeanset.size());
				Iterator<ObjectInstance> MBeansetIterator = MBeanset.iterator();

				while (MBeansetIterator.hasNext()) {
					ObjectInstance objectInstance = (ObjectInstance) MBeansetIterator.next();
					ObjectName objectName = objectInstance.getObjectName();
					String canonicalName = objectName.getCanonicalName();
					System.out.println("canonicalName : " + canonicalName);
					if (canonicalName.equals("Catalina:host=localhost,type=Cluster")) {
						// Get details of cluster MBeans
						System.out.println("Cluster MBeans Details:");
						System.out.println("=========================================");
						// getMBeansDetails(canonicalName);
						String canonicalKeyPropList = objectName.getCanonicalKeyPropertyListString();
					}
				}

				// ------------------------- system ----------------------
				ObjectName runtimeObjName = new ObjectName("java.lang:type=Runtime");
				System.out.println("厂商:" + (String) mbsc.getAttribute(runtimeObjName, "VmVendor"));
				System.out.println("程序:" + (String) mbsc.getAttribute(runtimeObjName, "VmName"));
				System.out.println("版本:" + (String) mbsc.getAttribute(runtimeObjName, "VmVersion"));
				Date starttime = new Date((Long) mbsc.getAttribute(runtimeObjName, "StartTime"));
				SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
				System.out.println("启动时间:" + df.format(starttime));

				Long timespan = (Long) mbsc.getAttribute(runtimeObjName, "Uptime");
//	            System.out.println("连续工作时间:" + test.formatTimeSpan(timespan)); 

				// ------------------------ JVM -------------------------
				// 堆使用率
				ObjectName heapObjName = new ObjectName("java.lang:type=Memory");
				MemoryUsage heapMemoryUsage = MemoryUsage.from((CompositeDataSupport) mbsc.getAttribute(heapObjName, "HeapMemoryUsage"));
				long maxMemory = heapMemoryUsage.getMax();// 堆最大
				long commitMemory = heapMemoryUsage.getCommitted();// 堆当前分配
				long usedMemory = heapMemoryUsage.getUsed();
				System.out.println("heap:" + (double) usedMemory * 100 / commitMemory + "%");// 堆使用率

				MemoryUsage nonheapMemoryUsage = MemoryUsage.from((CompositeDataSupport) mbsc.getAttribute(heapObjName, "NonHeapMemoryUsage"));
				long noncommitMemory = nonheapMemoryUsage.getCommitted();
				long nonusedMemory = heapMemoryUsage.getUsed();
				System.out.println("nonheap:" + (double) nonusedMemory * 100 / noncommitMemory + "%");

				ObjectName permObjName = new ObjectName("java.lang:type=MemoryPool,name=Perm Gen");
				MemoryUsage permGenUsage = MemoryUsage.from((CompositeDataSupport) mbsc.getAttribute(permObjName, "Usage"));
				long committed = permGenUsage.getCommitted();// 持久堆大小
				long used = heapMemoryUsage.getUsed();//
				System.out.println("perm gen:" + (double) used * 100 / committed + "%");// 持久堆使用率

				// -------------------- Session ---------------
				ObjectName managerObjName = new ObjectName("Catalina:type=Manager,*");
				Set<ObjectName> s = mbsc.queryNames(managerObjName, null);

				for (ObjectName obj : s) {
					System.out.println("应用名:" + obj.getKeyProperty("path"));
					ObjectName objname = new ObjectName(obj.getCanonicalName());
					System.out.println("最大会话数:" + mbsc.getAttribute(objname, "maxActiveSessions"));
					System.out.println("会话数:" + mbsc.getAttribute(objname, "activeSessions"));
					System.out.println("活动会话数:" + mbsc.getAttribute(objname, "sessionCounter"));
				}

				// ----------------- Thread Pool ----------------
				ObjectName threadpoolObjName = new ObjectName("Catalina:type=ThreadPool,*");
				Set<ObjectName> s2 = mbsc.queryNames(threadpoolObjName, null);

				for (ObjectName obj : s2) {
					System.out.println("端口名:" + obj.getKeyProperty("name"));
					ObjectName objname = new ObjectName(obj.getCanonicalName());
					System.out.println("最大线程数:" + mbsc.getAttribute(objname, "maxThreads"));
					System.out.println("当前线程数:" + mbsc.getAttribute(objname, "currentThreadCount"));
					System.out.println("繁忙线程数:" + mbsc.getAttribute(objname, "currentThreadsBusy"));
				}
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public static String formatTimeSpan(long span) {
		long minseconds = span % 1000;

		span = span / 1000;
		long seconds = span % 60;

		span = span / 60;
		long mins = span % 60;

		span = span / 60;
		long hours = span % 24;

		span = span / 24;
		long days = span;

		try (Formatter formatter = new Formatter()) {
			return formatter.format("%1$d天 %2$02d:%3$02d:%4$02d.%5$03d", days, hours, mins, seconds, minseconds).toString();
		}
	}
}
